package com.example.nirvana.dataStructure.algorithm;

import java.util.Stack;

/**
 * 用栈实现队列
 */
public class QueueByStack {
    private Stack<Integer> s1, s2;

    public QueueByStack() {
        s1 = new Stack<>();
        s2 = new Stack<>();
    }

    /**
     * 添加元素到队尾
     *
     * @param x 元素
     */
    public void push(int x) {
        s1.push(x);
    }

    /**
     * 返回队头元素
     *
     * @return
     */
    public int peek() {
        if (s2.isEmpty()) {
            // 把 s1 元素压入 s2
            while(!s1.isEmpty()) {
                s2.push(s1.pop());
            }
        }
        return s2.peek();
    }

    /**
     * 删除队头的元素并返回
     *
     * @return
     */
    public int pop() {
        //先调用 peek 保证 s2 非空
        peek();
        return s2.pop();
    }

    public boolean isEmpty() {
        return s1.isEmpty() && s2.isEmpty();
    }
}
